import torch
import add_hip

a = torch.ones(100, device='cuda')
b = torch.ones(100, device='cuda')

c = add_hip.add(a, b)
print(c)
